#include <linux/types.h>
#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/ide.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/gpio.h>
#include <linux/cdev.h>
#include <linux/device.h>
#include <linux/of.h>
#include <linux/of_address.h>
#include <linux/of_gpio.h>
#include <asm/mach/map.h>
#include <asm/uaccess.h>
#include <asm/io.h>

#define KEY_CNT 1     /* 设备号个数 */
#define KEY_NAME      "key"   /* 名字 */

/* 定义按键值 */
#define KEY0VALUE 0xF0   /* 按键值*/
#define INVAKEY   0x00  /* 无效的按键值 */


/* key设备结构体 */
struct key_dev{
    dev_t devid;           /* 设备号 */
    struct cdev cdev;      /* cdev */
    struct class *class;  /* 类 */
    struct device *device;   /* 设备 */
    int major;     /*  主设备号 */
    int minor;     /* 次设备号 */
    struct device_node *nd;    /* 设备节点 */
    int key_gpio;
    atomic_t keyvalue;         /* 按键值*/
};

struct key_dev keydev;

static int key_open(struct inode *inode, struct file *filp)
{
    filp->private_data = &keydev;
    keydev.nd = of_find_node_by_path("/key");
    if(keydev.nd == NULL){
        return -EINVAL;
    }

    keydev.key_gpio = of_get_named_gpio(keydev.nd, "key-gpio",0);
    if(keydev.key_gpio < 0) {
        printk("can't get key0\r\n");
        return -EINVAL;
    }
    printk("key_gpio = %d\r\n", keydev.key_gpio);
    /* 初始化key所使用的IO */
    gpio_request(keydev.key_gpio, "key0");
    gpio_direction_input(keydev.key_gpio);

    return 0;
}

static ssize_t key_read(struct file *filp, char __user *buf,
                        size_t cnt,loff_t *offt)
{
    int ret = 0;
    unsigned char value;
    struct key_dev *dev = filp->private_data;
    if(gpio_get_value(dev->key_gpio) == 0){
        while(!gpio_get_value(dev->key_gpio))
            ;
        atomic_set(&dev->keyvalue,KEY0VALUE);
    }else{
        atomic_set(&dev->keyvalue,INVAKEY);
    }
    value = atomic_long_read(&dev->keyvalue);
    ret =  copy_to_user(buf,&value,sizeof(value));
    return ret;
}

static struct file_operations key_fops = {
    .owner	= THIS_MODULE,
    .open	= key_open,
    .read	= key_read,
};

static __init mykey_init(void)
{
    /* 初始化源自变量 */
    atomic_set(&keydev.keyvalue,INVAKEY);
    /* 注册字符设备驱动 */
    /* 1、创建设备号 */
    if(keydev.major){   /* 如果定义了设备号 */
        keydev.devid = MKDEV(keydev.major,0);
        register_chrdev_region(keydev.devid,KEY_CNT,KEY_NAME);
    }else {    /* 没有定义设备号 */
        alloc_chrdev_region(&keydev.devid,0,KEY_CNT,KEY_NAME);
        keydev.major = MAJOR(keydev.devid);  /* 获取分配的主设备号 */
        keydev.minor = MINOR(keydev.devid);  /* 获取分配的次设备号 */
    }

    /* 2、初始化cdev */
    keydev.cdev.owner = THIS_MODULE;
    cdev_init(&keydev.cdev,&key_fops);

    /* 3、添加一个cdev */
    cdev_add(&keydev.cdev,keydev.devid,KEY_CNT);
    /* 4、创建类 */
    keydev.class = class_create(THIS_MODULE,KEY_NAME);
    if(IS_ERR(keydev.class)){
        return PTR_ERR(keydev.class);
    }

    /* 5、创建设备 */
    keydev.device = device_create(keydev.class,NULL,keydev.devid,NULL,KEY_NAME);
    if(IS_ERR(keydev.device)){
        return PTR_ERR(keydev.device);
    }
    
    return 0;
}

static __exit mykey_exit(void)
{
    //注销字符设备驱动
    gpio_free(keydev.key_gpio);
    cdev_del(&keydev.cdev);
    unregister_chrdev_region(keydev.devid,KEY_CNT);

    device_destroy(keydev.class,keydev.devid);
    class_destroy(keydev.class);
}

module_init(mykey_init);
module_exit(mykey_exit);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("zhangchaoliang");
